# coding:utf8
import warnings
import torch as t


class DefaultConfig(object):
    env = 'default'    # visdom 环境


    train_data_root = './data/train/'        # 训练集存放路径
    validate_data_root = './data/validation/'  # 验证集存放路径
    test_data_root = './data/test/'          # 测试集存放路径
    load_model_path = None                      # 加载模型的路径，为None代表不加载

    batch_size = 64   # batch size
    print_freq = 10  # print info every N batch

    result_file = 'result.csv'
    max_epoch = 1500
    lr = 0.01           # initial learning rate
    lr_decay = 0.95       # when val_loss increase, lr = lr*lr_decay
    weight_decay = 1e-4   # 损失函数


opt = DefaultConfig()